package service

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"io"
	corev1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"log"
)

var Pod pod

type pod struct{}

//定义列表返回内容 Items是pod元素列表 total是元素数量
type PodsResp struct {
	Total int          `json:"total"`
	Items []corev1.Pod `json:"items"`
}

//类型转换方法 corev1.Pod -> DataCell ,
func (p *pod) toCell(pods []corev1.Pod) []DataCell {
	cell := make([]DataCell, len(pods))
	for i := range pods {
		cell[i] = podCell(pods[i])
	}
	return cell
}

//类型转换方法  DataCell -> corev1.Pod
func (p *pod) fromCells(cells []DataCell) []corev1.Pod {
	pods := make([]corev1.Pod, len(cells))
	for i := range cells {
		// cells[i].(podCell)是将DataCell类型转换成podCell
		pods[i] = corev1.Pod(cells[i].(podCell))

	}
	return pods
}

//获取pod列表 支持过滤 排序 分页
func (p *pod) GetPods(filterName, Namespace string, limit, page int) (podsResp *PodsResp, err error) {
	//获取podList类型的pod列表
	//context.TODO()用于声明一个空的context上下文，用于List方法内设置这个请求的超时(源码），这里的常用用法
	//metav1.Listoptions{0用于过滤List数据，如使用1abe1， field等
	//kubect1 get services --all-namespaces --field-seletor metadat a.namespace != default
	podlist, err := K8s.ClientSet.CoreV1().Pods(Namespace).List(context.TODO(), metav1.ListOptions{})
	if err != nil {
		log.Printf("[ Pod ]资源列表获取失败：%s", err)
		return nil, errors.New("获取Pod列表失败" + err.Error())
	}

	//实例化dataSelector结构体，组装数据
	selectableData := &dataSelector{
		GenericDataList: p.toCell(podlist.Items),
		DataSelectQuery: &DataSelectQuery{
			FilterQuery: &FilterQuery{
				Name: filterName,
			},
			PaginateQuary: &PaginateQuary{
				Limit: limit,
				Page:  page,
			},
		},
	}

	//先过滤
	filtered := selectableData.Filter()
	total := len(filtered.GenericDataList)
	//先排序再分页
	data := filtered.Sort().Paginate()

	pods := p.fromCells(data.GenericDataList)

	return &PodsResp{
		Total: total,
		Items: pods,
	}, nil

}

//获取pod详情
func (p *pod) GetPodDetail(podName, namespace string) (pod *corev1.Pod, err error) {
	podinfo, err := K8s.ClientSet.CoreV1().Pods(namespace).Get(context.TODO(), podName, metav1.GetOptions{})
	if err != nil {
		log.Printf("获取pod详情失败：%s", err)
		return nil, errors.New("获取pod详情失败：%s" + err.Error())
	}
	return podinfo, nil
}

//删除pod
func (p *pod) DeletePod(podName, namespace string) (err error) {
	err = K8s.ClientSet.CoreV1().Pods(namespace).Delete(context.TODO(), podName, metav1.DeleteOptions{})
	if err != nil {
		log.Printf("删除pod失败：%s", err)
		return errors.New("删除pod失败:" + err.Error())
	}
	return nil
}

//更新pod
//content参数是请求中传入的pod对象的json数据
func (p *pod) UpDatePod( namespace, content string) (err error) {
	var pod = &corev1.Pod{}
	//反序列化为pod对象
	err = json.Unmarshal([]byte(content), pod)
	if err != nil {
		log.Printf("反序列化失败：%s", err)
		return errors.New("反序列化失败:" + err.Error())
	}
	//更新pod
	_, err = K8s.ClientSet.CoreV1().Pods(namespace).Update(context.TODO(), pod, metav1.UpdateOptions{})
	if err != nil {
		log.Printf("pod 更新失败：%s", err)
		return errors.New("pod 更新失败:" + err.Error())
	}
	return nil
}

//获取pod中的容器名
func (p *pod) GetPodContainer(podName, namespace string) (containers []string, err error) {
	//获取pod详情
	pod, err := p.GetPodDetail(podName, namespace)
	if err != nil {
		return nil, err
	}
	//从pod中拿到容器名
	for _, container := range pod.Spec.Containers {
		containers = append(containers, container.Name)
	}
	return containers, nil
}

//获取pod中的日志
func (p *pod) GetPodLog(containerName, podName, namespace string) (loginfo string, err error) {
	//设置日志的配置，容器名，tail的行数
	lineLimit := int64(500)
	option := &corev1.PodLogOptions{
		Container: containerName,
		TailLines: &lineLimit,
	}
	//获取request实例
	req := K8s.ClientSet.CoreV1().Pods(namespace).GetLogs(podName,option)
	//发起request请求 返回一个io.readcloser类型 （等同于response.body）
	podLogs, err := req.Stream(context.TODO())
	if err != nil {
		log.Printf("pod 日志获取失败：%s", err)
		return "",errors.New("pod 日志获取失败：%s" + err.Error())
	}
	defer podLogs.Close()
	//将response body写入到缓冲区 转成string返回
	buf := new(bytes.Buffer)
	_,err = io.Copy(buf,podLogs)
	if err != nil {
		log.Printf("pod 日志写入到缓冲区失败：%s", err)
		return "",errors.New("pod 日志写入到缓冲区失败：%s" + err.Error())
	}
	return buf.String(),nil
}

type PodsNp struct {
	Namespace string
	PodNum int
}

//获取每个namespace的pod数量
func (p *pod) GetPodNumPerNp() (podsNps []*PodsNp,err error){
	//获取namespace列表
	namespaceList, err := K8s.ClientSet.CoreV1().Namespaces().List(context.TODO(),metav1.ListOptions{})
	if err != nil {
		return nil,err
	}
	for _ ,namespace := range namespaceList.Items{
		//获取pod列表
		postList, err := K8s.ClientSet.CoreV1().Pods(namespace.Name).List(context.TODO(),metav1.ListOptions{})
		if err != nil {
			return nil,err
		}
		//组装数据
		podsNp := &PodsNp{
			Namespace: namespace.Name,
			PodNum: len(postList.Items),
		}
		//添加到podsNps数组中
		podsNps = append(podsNps,podsNp)
	}
	return podsNps,nil
}
